Skip to content

Fix sagemaker-core unit test and sagemaker-serve integ test#5509

Merged
mollyheamazon merged 2 commits intoaws:masterfrom
mollyheamazon:master
Jan 22, 2026
Merged

Fix sagemaker-core unit test and sagemaker-serve integ test#5509
mollyheamazon merged 2 commits intoaws:masterfrom
mollyheamazon:master

Conversation

@mollyheamazon
Copy link
Contributor

@mollyheamazon mollyheamazon commented Jan 22, 2026 •
edited
Loading

Issue #, if available:
There are test failures in #5499 and #5489. The failures do not come from the code changes of these PR, but rather require fixes due to the recent commit of DJL 0.36.0 release: #5486

Description of changes:

For unit test in sagemaker-core

FAILED tests/unit/deserializers/test_base_deserializers.py::test_pandas_deserializer_json
FAILED tests/unit/deserializers/test_base_deserializers.py::test_pandas_deserializer_csv
FAILED tests/unit/image_uris/test_djl.py::test_djl_lmi_config_for_framework_has_all_regions[djl-lmi]
FAILED tests/unit/image_uris/test_djl.py::test_djl_lmi_config_for_framework_has_all_regions[djl-tensorrtllm]
  • Add known missing combinations of DJL images
  • Add pytest autouse fixture to prevent pandas mock leak from polluting other tests in Python 3.11/3.12

For integ test failure in sagemaker-serve

FAILED tests/integ/test_huggingface_integration.py::test_huggingface_build_deploy_invoke_cleanup - botocore.errorfactory.ValidationError: An error occurred (ValidationError) when calling the InvokeEndpoint operation: Endpoint hf-test-endpoint-d33a5c59 of account 729646638167 not found.
  • Update hugging face model id that is compatible with DJL 0.36.0 vLLM

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.

@mollyheamazon mollyheamazon merged commit 4e55189 into aws:master Jan 22, 2026
14 of 21 checks passed
This was referenced Jan 22, 2026
Merged
Merged
@mollyheamazon mollyheamazon mentioned this pull request Jan 26, 2026
Merged
access2rohit added a commit to access2rohit/sagemaker-python-sdk that referenced this pull request Feb 2, 2026
@access2rohit
added missing framework:version:region combinations to fix failing DJ…
c6cff06 
…L tests ref aws#5509
nargokul added a commit that referenced this pull request Feb 2, 2026
@access2rohit @nargokul
change: Updates for DJL 0.36.0 release (#5486) (#5528)
2a29584 
* change: Updates for DJL 0.36.0 release (#5486)

Signed-off-by: Rohit Kumar Srivastava <srivastava.141@osu.edu>
Co-authored-by: Gokul Anantha Narayanan <166456257+nargokul@users.noreply.github.com>

* added missing framework:version:region combinations to fix failing DJL tests ref #5509

---------

Signed-off-by: Rohit Kumar Srivastava <srivastava.141@osu.edu>
Co-authored-by: Gokul Anantha Narayanan <166456257+nargokul@users.noreply.github.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

@aviruthen aviruthen aviruthen approved these changes

@pintaoz-aws pintaoz-aws pintaoz-aws approved these changes

@rsareddy0329 rsareddy0329 rsareddy0329 approved these changes

Assignees

No one assigned

Labels

None yet

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

4 participants

@mollyheamazon @aviruthen @pintaoz-aws @rsareddy0329